package solution.design;

import solution.Solution_4tree;

import java.util.Stack;

/**
 * @author zhangmin
 * @create 2021-10-21 14:10
 *
 * 设计问题--232. 用栈实现队列
 * 请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作（push、pop、peek、empty）：
 */
public class MyQueue {
    private Stack<Integer> stack1,stack2;

    public MyQueue() {
        stack1=new Stack<>();
        stack2=new Stack<>();
    }

    //将元素 x 推到队列的末尾
    public void push(int x) {
        stack1.push(x);
    }

    //从队列的开头移除并返回元素
    public int pop() {
        // 先调用 peek 保证 s2 非空
        peek();
        return stack2.pop();
    }

    //返回队列开头的元素:当 s2 为空时，可以把 s1 的所有元素取出再添加进 s2，这时候 s2 中元素就是先进先出顺序了。
    public int peek() {
        if (stack2.isEmpty()) {
            while (!stack1.isEmpty())
                stack2.push(stack1.pop());
        }
        return stack2.peek();
    }

    //如果队列为空，返回 true ；否则，返回 false
    public boolean empty() {
        return stack1.isEmpty()&& stack2.isEmpty();
    }

}
